package cn.iocoder.yudao.module.product.dal.mysql.property;

import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.product.controller.admin.property.vo.value.ProductPropertyValuePageReqVO;
import cn.iocoder.yudao.module.product.dal.dataobject.property.ProductPropertyDO;
import cn.iocoder.yudao.module.product.dal.dataobject.property.ProductPropertyValueDO;
import cn.iocoder.yudao.module.product.enums.property.PropertyValueTypeEnum;
import com.github.yulichang.wrapper.MPJLambdaWrapper;
import org.apache.ibatis.annotations.Mapper;

import java.util.Collection;
import java.util.List;

@Mapper
public interface ProductPropertyValueMapper extends BaseMapperX<ProductPropertyValueDO> {

    default List<ProductPropertyValueDO> selectListByPropertyId(Collection<Long> propertyIds) {
        return selectList(new LambdaQueryWrapperX<ProductPropertyValueDO>()
                .inIfPresent(ProductPropertyValueDO::getPropertyId, propertyIds));
    }

    default ProductPropertyValueDO selectByName(Long propertyId, String name) {
        return selectOne(new LambdaQueryWrapperX<ProductPropertyValueDO>()
                .eq(ProductPropertyValueDO::getPropertyId, propertyId)
                .eq(ProductPropertyValueDO::getName, name));
    }

    default void deleteByPropertyId(Long propertyId) {
        delete(new LambdaQueryWrapperX<ProductPropertyValueDO>()
                .eq(ProductPropertyValueDO::getPropertyId, propertyId));
    }

    default PageResult<ProductPropertyValueDO> selectPage(ProductPropertyValuePageReqVO reqVO) {
        return selectPage(reqVO, new LambdaQueryWrapperX<ProductPropertyValueDO>()
                .eqIfPresent(ProductPropertyValueDO::getPropertyId, reqVO.getPropertyId())
                .likeIfPresent(ProductPropertyValueDO::getName, reqVO.getName())
                .orderByDesc(ProductPropertyValueDO::getId));
    }

    default Integer selectCountByPropertyId(Long propertyId) {
        return selectCount(ProductPropertyValueDO::getPropertyId, propertyId).intValue();
    }

    // ========== 新增 =========

    default ProductPropertyValueDO selectByNameAndSkuId(Long skuId, Long propertyId, String name) {
        return selectOne(new LambdaQueryWrapperX<ProductPropertyValueDO>()
                .eq(ProductPropertyValueDO::getPropertyId, propertyId)
                .eq(ProductPropertyValueDO::getName, name)
                .eq(ProductPropertyValueDO::getSkuId, skuId));
    }

    default ProductPropertyValueDO selectByNameAndSpuId(Long spuId, Long propertyId, String name) {
        return selectOne(new LambdaQueryWrapperX<ProductPropertyValueDO>()
                .eq(ProductPropertyValueDO::getPropertyId, propertyId)
                .eq(ProductPropertyValueDO::getName, name)
                .eq(ProductPropertyValueDO::getType, PropertyValueTypeEnum.SPU.getValue())
                .eq(ProductPropertyValueDO::getSpuId, spuId));
    }

    default List<ProductPropertyValueDO> selectBySkuIds(Collection<Long> skuIds) {
        return selectList(new LambdaQueryWrapperX<ProductPropertyValueDO>()
                .in(ProductPropertyValueDO::getSkuId, skuIds));
    }

    default List<ProductPropertyValueDO> selectBySpuIds(PropertyValueTypeEnum type, Collection<Long> spuIds) {
        return selectList(new LambdaQueryWrapperX<ProductPropertyValueDO>()
                .eq(ProductPropertyValueDO::getType, type.getValue())
                .in(ProductPropertyValueDO::getSpuId, spuIds));
    }

    default List<ProductPropertyValueDO> selectListByPropertyNameGroupByValue(String propertyName){
        final MPJLambdaWrapper<ProductPropertyValueDO> wrapper = new MPJLambdaWrapper<ProductPropertyValueDO>()
                .selectAll(ProductPropertyValueDO.class)
                .innerJoin(ProductPropertyDO.class, ProductPropertyDO::getId, ProductPropertyValueDO::getPropertyId)
                .eq(ProductPropertyDO::getName, propertyName)
                .eq(ProductPropertyValueDO::getDeleted, false)
                .groupBy(ProductPropertyValueDO::getName);

        final List<ProductPropertyValueDO> valueDOList = selectJoinList(ProductPropertyValueDO.class, wrapper);
        return valueDOList;
    }

    /**
     * 根据属性id和属性值查询，并使用propertyValue group by
     * @param type
     * @param propertyId
     * @param propertyValueList
     * @return
     */
    default List<ProductPropertyValueDO> selectGroupBySpuId(PropertyValueTypeEnum type,
                                                            Long propertyId,
                                                            List<String> propertyValueList) {
        return selectList(new LambdaQueryWrapperX<ProductPropertyValueDO>()
                .eq(ProductPropertyValueDO::getType, type.getValue())
                .eq(ProductPropertyValueDO::getPropertyId, propertyId)
                .in(ProductPropertyValueDO::getName, propertyValueList)
                .groupBy(ProductPropertyValueDO::getSpuId)
                .having("COUNT(DISTINCT name) = " + propertyValueList.size())
        );
    }

}
